package com.aaa.boot.util;

import com.aaa.boot.pojo.dto.PermDTO;

import java.util.ArrayList;
import java.util.List;

public class PermissionUtil {
    public static List<PermDTO> getPermission(List<PermDTO> permissionDtos){
        ArrayList<PermDTO> data = new ArrayList<>();
        for(PermDTO permissionDto:permissionDtos){
            if(permissionDto.getPid()==0){
                data.add(permissionDto);
            }else{
                Integer pid = permissionDto.getPid();
                for(PermDTO datum:data){
                    if(datum.getId().equals(pid)){
                        datum.getSubMenu().add(permissionDto);
                    }
                }
            }
        }

        return data;
    }
}
